package vue.FenetrePartie;

import javax.swing.ImageIcon;
import javax.swing.JButton;
public class Unite {
	
	/**
	 * 
	 */
	private modele.Unite.Unite unite;
	private JButton button;

	/**
	 * 
	 * @param unite
	 * Cette classe est comme la classe Batiment de la vue, elle va nous permettre de rendre le code plus clair en évitant de répéter toujours les même lignes 
	 * pour la création d'une unité sur la map.
	 */
	public Unite(modele.Unite.Unite unite)
	{
		this.unite = unite;
		this.button = new JButton();
		if(this.unite!=null) this.setButton();
 	}



	public modele.Unite.Unite getUnite() {
		return this.unite;
	}

	public void setUnite(modele.Unite.Unite unite) {
		this.unite = unite;
		this.setButton();
		
	}

	public JButton getButton() {
		return button;
	}
	
	public void setButton()
	{
		ImageIcon image = new ImageIcon("images/finies/"+this.unite.getClass().getName()+".png"); 
		this.button.setIcon(image);
		this.button.setBorderPainted(false);  
		this.button.setFocusPainted(false);  
		this.button.setContentAreaFilled(false);
	}
}
